#!/bin/bash

set -e -o pipefail

BASE=$(dirname "$(readlink -fm "$0")")
DISTRIB=$(cat /etc/lsb-release | grep "DISTRIB_ID" | awk -F "=" '{print $2}')
DISTRIB_RELEASE=$(cat /etc/lsb-release | grep "DISTRIB_RELEASE" | awk -F "=" '{print $2}')
KHADAS_TOOL_NAME="rk-burn-tool"
ANDROID_TOOL_CONFIG="$BASE/tools/Linux_Upgrade_Tool/config.ini"
INSTALL_DIR="/usr/local/bin"

RULE=

RED='\033[0;31m'
RESET='\033[m'

error_msg() {
	echo -e ${RED}Error:${RESET} $1
}

echo ""
echo "==============================================="
echo ""
echo "Host PC: $DISTRIB $DISTRIB_RELEASE"
echo ""
echo "==============================================="
echo ""

if [ "$DISTRIB" != "Ubuntu" ]; then
	error_msg "This flash tool just support Ubuntu now, other distributions haven't been verified!"
	exit 1
fi

echo "Removing USB rules..."

if [[ "$DISTRIB_RELEASE" =~ "16" || "$DISTRIB_RELEASE" =~ "18" || "$DISTRIB_RELEASE" =~ "20" ]]; then
	RULE="/etc/udev/rules.d/80-rockchip-usb.rules"
else
	error_msg "Ubuntu $DISTRIB_RELEASE haven't been verified!"
	exit 1
fi

sudo rm -f $RULE
sudo udevadm control --reload-rules

echo "Removing flash-tool..."
sudo rm -f $INSTALL_DIR/$KHADAS_TOOL_NAME

echo "Removing configuration..."
if [ -d ~/.config/upgrade_tool ]; then
	rm -r ~/.config/upgrade_tool
fi

echo "Done!"
